package io.xccit.deepseek.service;

import io.xccit.deepseek.entity.ChatEntity;
import org.springframework.ai.chat.ChatResponse;
import reactor.core.publisher.Flux;

import java.util.Map;

/**
 * @author CH_ywx
 * @version 1.0
 * @create 2025/4/7
 * <p>Ollama业务层接口</p>
 */
public interface OllamaService {

    /**
     * 聊天
     * @param message 消息
     * @return 聊天结果
     */
    Map<String, Object> chat(String message);

    /**
     * 聊天流式返回
     * @param message 消息
     * @return 聊天结果
     */
    Flux<ChatResponse> chatStream(String message);

    /**
     * 聊天医生
     * @param chatEntity
     */
    void chatDoctor(ChatEntity chatEntity);
}
